#!/usr/bin/env python3
"""Mental Health Analysis Script
Author: Juan Moises de la Serna
"""
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt

df = pd.read_csv('mental_health_extended.csv')
print(df.describe())

corr = df[['depression_pct', 'anxiety_pct', 'suicide_rate_100k']].corr()
sns.heatmap(corr, annot=True, cmap='coolwarm')
plt.title('Mental Health Indicators Correlation')
plt.tight_layout()
plt.savefig('mental_health_corr.png', dpi=150)
